package com.mydemo.common.utils;

import com.mydemo.common.Constant;
import com.mydemo.common.pojo.LoginUser;
import com.mydemo.common.pojo.Role;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;

import java.util.List;

/**
 * Security工具类
 * @author Last_Island
 * @version 1.0
 */
public class SecurityUtils {

    /**
     * 获取当前用户
     * @return
     */
    public static LoginUser getLoginUser()
    {
        Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();
        if (principal instanceof LoginUser)
            return (LoginUser) principal;
        return null;
    }

    /**
     * 判断是否为管理员
     * @return
     */
    public static boolean isAdmin()
    {
        LoginUser loginUser = getLoginUser();
        List<Role> roles = loginUser.getRoles();
        for (Role role : roles) {
            if (role.getRoleKey().equals(Constant.ADMIN_ROLE_KEY))
                return true;

        }
        return false;
    }

    /**
     * 生成BCryptPasswordEncoder密码
     *
     * @param password 密码
     * @return 加密字符串
     */
    public static String encryptPassword(String password)
    {
        BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
        return passwordEncoder.encode(password);
    }

    /**
     * 判断密码是否相同
     *
     * @param rawPassword 真实密码
     * @param encodedPassword 加密后字符
     * @return 结果
     */
    public static boolean matchesPassword(String rawPassword, String encodedPassword)
    {
        BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
        return passwordEncoder.matches(rawPassword, encodedPassword);
    }
}
